import torch
from torch import nn

class Ah(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self,input):
        output = input + 1
        return output

ah = Ah()
x = torch.tensor(1.0)
output = ah(x)
print(output)